package com.aotain.jupiter.flink;

import java.util.regex.Matcher;
import java.util.regex.Pattern;

import org.apache.commons.codec.binary.Base64;
import org.apache.flink.api.common.functions.MapFunction;
import org.apache.flink.api.java.tuple.Tuple2;

import com.aotain.apollo.IPDatabase;
import com.aotain.apollo.IPSearcher;
import com.aotain.common.CommonFunction;

public class SourceDataStat implements MapFunction<String,Tuple2<String,String>>{

	/** 
	 * @Fields serialVersionUID : TODO(用一句话描述这个变量表示什么) 
	 */ 
	private static final long serialVersionUID = -5887489583949779684L;

	String sourceAreaCountry = null;
	String sourceAreaCity = null;
	String sourceAreaProvince = null;
	String sourceAreaName = null;
	String destAreaCountry = null;   
	String destAreaCity = null;
	String destAreaName = null;
	String destAreaProvince = null;
	String gisstr = null;
	String gisstrBase64 = null;
	String Sourcelon = null;
	String Sourcelat = null;
	String Destlon = null;
	String Destlat = null;
	String sourceAreaId = null;
	String sourceAreaProvinceId = null;
	String sourceAreaCityId = null;
	String destAreaId = null;
	String destAreaProvinceId = null;
	String destAreaCityId = null;
	String areaName_s = null;
	String areaName_d = null;
	String dip = null;
	String sip = null;
	String userAgent = "";
	String cookie = "";
	Long[] IPs = null;
	String dipL = null;
	String sipL = null;
	String dport = null;
	String sport = null;
	String filterIPs = null;
	String domainname = null;
	String url = null;
	String type = null;
	String accesstime = null;
	String idcno = null;
	String delaytime = null;
	long upStreamOctets = 0;
	long upStreamPacket = 0;
	long dnStreamOctets = 0;
	long dnStreamPacket = 0;
	Integer status ;
	String dsip = null;
	String refer = "";
	String dbJson = "";
	

	public SourceDataStat(){}

	public SourceDataStat(String dbJson) {
		this.dbJson = dbJson;
		
	}

	@Override
	public Tuple2<String,String> map(String in) throws Exception {


		if(in != null && !"".equals(in)) {
			String[] splits = in.split("\\|",-1);
			if(splits.length <= 1)
				return new Tuple2<String,String>("","");

			String userAccount = splits[1];
			String ipRex = "([1-9]|[1-9]\\d|1\\d{2}|2[0-4]\\d|25[0-5])(\\.(\\d|[1-9]\\d|1\\d{2}|2[0-4]\\d|25[0-5])){3}";   
			Pattern pattern = Pattern.compile(ipRex);   
			Matcher matcher = pattern.matcher(userAccount);  
			if(matcher.matches()) {
				if("1".equals(splits[22]) && "6".equals(splits[7]) && splits.length > 42) {//http
					dipL = splits[9];
					dport = splits[10];
					sipL = splits[12];
					sport = splits[13];
					domainname = splits[32];
					url = domainname + "/" + splits[28];
					type = splits[7]; 
					accesstime= splits[14];
					idcno = "911";
					dnStreamOctets = Long.valueOf(splits[17]);
					upStreamOctets = Long.valueOf(splits[18]);
					dnStreamPacket = Long.valueOf(splits[19]);
					upStreamPacket = Long.valueOf(splits[20]);
					status = Integer.valueOf(splits[23]);
					userAgent = splits[26];
					cookie = splits[42];
					refer = splits[37];
				}else{//非Http
					dipL = splits[9];
					dport = splits[10];
					sipL = splits[12];
					sport = splits[13];
					domainname = "";
					url = "";
					type = splits[7]; 
					accesstime= splits[14];
					idcno = "911";
					dnStreamOctets = Long.valueOf(splits[17]);
					upStreamOctets = Long.valueOf(splits[18]);
					dnStreamPacket = Long.valueOf(splits[19]);
					upStreamPacket = Long.valueOf(splits[20]);
					status = Integer.valueOf(splits[23]);
				}
				delaytime = 0 + "";
				long currenttime = System.currentTimeMillis();
				if(accesstime != null && !"".equals(accesstime)) {
					delaytime = currenttime/1000 - Long.parseLong(accesstime.trim()) + "";
				}

				if(!"".equals(dipL) && !"".equals(dport) && !"".equals(sipL) && !"".equals(sport) ) {
					dip = CommonFunction.int2ip(Long.parseLong(dipL));
					sip = CommonFunction.int2ip(Long.parseLong(sipL));

					//					System.out.println(dip+ "|" +dport+ "|" +sip+ "|" +sport+ "|" +domainname+ "|" +url+ "|" +type+ "|" +delaytime+ "|" +idcno+ "|" +upStreamOctets+ "|" +upStreamPacket+ "|" +status+ "|" +userAgent+ "|" +cookie+ "|" +dnStreamOctets+ "|" +dnStreamPacket+ "|" +refer);
					long sipLong = Long.parseLong(sipL);
					if((sipLong >= 167772160 && sipLong <= 184549375) 
							|| (sipLong >= 2886729728L && sipLong <= 2887778303L)
							|| (sipLong >= 3232235520L &&sipLong <= 3232301055L)) {

						IPSearcher ipsearcher = IPSearcher.getInstance(dbJson);
						IPDatabase SourceArea = ipsearcher.SearchIPDB(sip);
						IPDatabase DestArea = ipsearcher.SearchIPDB(dip);

						if(DestArea != null) {
							destAreaCountry = DestArea.getCountryName();
							destAreaCity = DestArea.getCityName();
							//						sourceAreaProvince = SourceArea.getProviceName();
							if(destAreaCity != null) {
								destAreaName  = destAreaCity;
								destAreaId = DestArea.getCityID()+"";
							} else {
								destAreaName = destAreaCountry;
								destAreaId = DestArea.getCountryID()+"";
							}
							destAreaCityId = sourceAreaId;
							Destlon = String.valueOf(DestArea.getLon());
							Destlat = String.valueOf(DestArea.getLat());

							if(!"".equals(DestArea.getProviceID()) && DestArea.getProviceID() !=0 ) {
								destAreaId = DestArea.getProviceID()+"";
								areaName_d = DestArea.getProviceName().replace("省", "");
								areaName_d = areaName_d.replace("市", "");
							} else {
								areaName_d = DestArea.getCountryName();
								destAreaId = DestArea.getCountryID()+"";
							}
							destAreaProvinceId = destAreaId;
						}   

						if(SourceArea != null) {    
							sourceAreaCountry = SourceArea.getCountryName();
							sourceAreaCity = SourceArea.getCityName();
							if(sourceAreaCity != null) {
								sourceAreaName = sourceAreaCity;
							} else {   
								sourceAreaName = sourceAreaCountry;
							}
							Sourcelon = String.valueOf(SourceArea.getLon());//经度
							Sourcelat = String.valueOf(SourceArea.getLat());//纬度
						}  

						if(destAreaName != null && Destlon != null && Destlat != null && sourceAreaName != null && Sourcelon != null && Sourcelat != null && sourceAreaCountry != null) {
							gisstr = destAreaName + "#" + Destlon + "," + Destlat + "#" + sourceAreaName + "#" + Sourcelon + "," + Sourcelat + "#" + destAreaCountry + "#" + destAreaId + "#" +areaName_d + "#" + destAreaCityId + "#" + destAreaProvinceId + "#" + sourceAreaCountry;
							gisstrBase64 = new String(Base64.encodeBase64(gisstr.getBytes()));
							return new Tuple2<String,String>(sip,dip+ "|" +dport+ "|" +sip+ "|" +sport+ "|" +gisstrBase64+ "|" +domainname+ "|" +url+ "|" +type+ "|" +delaytime+ "|" +idcno+ "|" +upStreamOctets+ "|" +upStreamPacket+ "|" +status+ "|" +userAgent+ "|" +cookie+ "|" +dnStreamOctets+ "|" +dnStreamPacket+ "|" +refer);
						}
					}
				}
			}
		}
		return new Tuple2<String,String>("","");
	}
}
